import time
from generator import generate_data
from alg import shell_sort


def test_shell_sort():
    # 生成测试数据
    data = generate_data(10)
    original_data = data.copy()

    print("排序前的数据:", original_data)

    # 记录开始时间
    start_time = time.time()

    # 执行排序算法
    sorted_data = shell_sort(data)

    # 记录结束时间
    end_time = time.time()

    print("排序后的数据:", sorted_data)

    # 检查测试用例是否正确
    is_correct = sorted(original_data) == sorted_data

    # 打印测试结果
    print("测试用例是否正确:", is_correct)
    print("运行时间:", end_time - start_time, "秒")


if __name__ == "__main__":
    test_shell_sort()